package arrays;

import java.util.PriorityQueue;

/**
 * @author pengfei.hpf
 * @date 2020/3/27
 * @verdion 1.0.060.
 * 第k个排列
 * 给出集合 [1,2,3,…,n]，其所有元素共有 n! 种排列。
 *
 * 按大小顺序列出所有排列情况，并一一标记，当 n = 3 时, 所有排列如下：
 *
 * "123"
 * "132"
 * "213"
 * "231"
 * "312"
 * "321"
 * 给定 n 和 k，返回第 k 个排列。
 *
 * 说明：
 *
 * 给定 n 的范围是 [1, 9]。
 * 给定 k 的范围是[1,  n!]。
 * 示例 1:
 *
 * 输入: n = 3, k = 3
 * 输出: "213"
 * 示例 2:
 *
 * 输入: n = 4, k = 9
 * 输出: "2314"
 */
public class GetPermutation {
    public String getPermutation(int n, int k) {
        int[] nums = new int[n];
        for(int i = 1; i <= n; i++){
            nums[i-1] = i;
        }
        backtrack(0, nums, n, k);
        return queue.poll().toString();
    }

    private PriorityQueue<Integer> queue = new PriorityQueue<>((o1, o2) -> (o2 - o1));
    private void backtrack(int start, int[] nums, int n, int k){
        if(start == n-1){
            int num = 0;
            for(int i: nums){
                num = num *10 + i;
            }
            System.out.println(num);
            if(queue.size() >= k && queue.peek() > num){
                queue.poll();
                queue.add(num);
            } else if(queue.size() < k){
                queue.add(num);
            }
            return;
        }
        for(int i = start; i < n; i ++){
            swap(nums, start, i);
            backtrack(start + 1, nums, n, k);
            swap(nums, start, i);
        }
    }

    private void swap(int[] nums, int i, int j){
        int t = nums[i];
        nums[i] = nums[j];
        nums[j] = t;
    }

    public static void main(String[] args){
        System.out.println(new GetPermutation().getPermutation(3,6));
    }
}
